tensor([[[0.3100, 0.3100],
         [0.3100, 0.3100]],

        [[0.3000, 0.3000],
         [0.3000, 0.3000]],

        [[0.3900, 0.3900],
         [0.3900, 0.3900]]], grad_fn=<RoundBackward1>)